from modelscope import snapshot_download

# 指定模型名称
model_name = "shakechen/Llama-2-7b-hf"

# 指定下载路径（绝对路径
custom_path = "/home/mht/FLAP/aimodels3"

# 下载模型
model_dir = snapshot_download(
    model_name,  # 模型名称
    cache_dir=custom_path,  # 指定下载路径
    revision="master"  # 可选：指定模型版本（默认master）
)

print(f"模型已下载到：{model_dir}")
